#coding: utf-8
__author__ = 'Sail'

from adaboost import  *

# dataMat, classLabels = loadSimpleData()
# D = mat(ones((5,1))/5)
# print buildStump(dataMat,classLabels,D)
#
# classifierArray = adaBoostTrainDS(dataMat, classLabels, 9)
# print classifierArray
#
# print adaClassify([0,0],classifierArray)

datArr, labelArr = loadDataSet('horseColicTraining2.txt')
classifierArray,aggClassEst = adaBoostTrainDS(datArr,labelArr,10)
plotROC(aggClassEst.T,labelArr)